import websocket
try:
    import thread
except ImportError:
    import _thread as thread
import time

def on_message(ws, message):
    print(message)

def on_error(ws, error):
    print(error)

def on_close(ws):
    print("### closed ###")

def on_open(ws):
    def run(*args):
        ws.send("ok")
        #ws.close()
        print("thread terminating...")
    thread.start_new_thread(run, ())


if __name__ == "__main__":
    # websocket.enableTrace(True)
    # ws = websocket.WebSocketApp("ws://127.0.0.1:8099",
    #                             on_message = on_message,
    #                             on_error = on_error,
    #                             on_close = on_close)
    # ws.on_open = on_open
    # ws.run_forever()
    '''短链接测试'''
    ws = websocket.create_connection("ws://127.0.0.1:8099")
    print("Sending 'Hello, World'...")
    ws.send("Hello, World")
    print("Sent")
    print("Receiving...")
    result =  ws.recv()
    print("Received '%s'" % result)
    ws.close()
